package com.zhl.study.controller;

import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.ocr.v20181119.OcrClient;
import com.tencentcloudapi.ocr.v20181119.models.FormulaOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.FormulaOCRResponse;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.codec.binary.Base64;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;

/**
 * @description 腾讯云ocr
 * @Author: z156940l@outlook.com
 * @className ImageOcrController
 * @Date: 2023/5/18 21:32
 */
@RestController
@RequestMapping("/tctOcr")
public class ImageOcrController {

    private String secretId = "AKID9DSdfXDwY1JPL9nItcIyY1SmEessGWZN";
    private String secretKey = "f7LNDvPgmmaCdjgT9tqkYEk4CC3jLvcG";

    @PostMapping(value = "analysis")
    @ApiOperation(httpMethod = "POST", value = "腾讯云ocr识别数学题")
    public String picAnalysis(@RequestParam("file") MultipartFile file) throws TencentCloudSDKException, IOException {

        // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
        // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
        // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
        Credential cred = new Credential(secretId, secretKey);
        // 实例化一个http选项，可选的，没有特殊需求可以跳过
        HttpProfile httpProfile = new HttpProfile();
        httpProfile.setEndpoint("ocr.tencentcloudapi.com");
        // 实例化一个client选项，可选的，没有特殊需求可以跳过
        ClientProfile clientProfile = new ClientProfile();
        clientProfile.setHttpProfile(httpProfile);
        // 实例化要请求产品的client对象,clientProfile是可选的
        OcrClient client = new OcrClient(cred, "ap-beijing", clientProfile);
        // 实例化一个请求对象,每个接口都会对应一个request对象
        FormulaOCRRequest req = new FormulaOCRRequest();
        String base64EncoderImg = Base64.encodeBase64String(file.getBytes());
        req.setImageBase64(base64EncoderImg);
        // 返回的resp是一个FormulaOCRResponse的实例，与请求对象对应
        FormulaOCRResponse resp = client.FormulaOCR(req);
        // 输出json格式的字符串回包
        System.out.println(FormulaOCRResponse.toJsonString(resp));
        return null;
    }
}
